package com.atguigu.tingshu.common.login;

import com.atguigu.tingshu.common.constant.RedisConstant;
import com.atguigu.tingshu.common.execption.GuiguException;
import com.atguigu.tingshu.common.result.ResultCodeEnum;
import com.atguigu.tingshu.common.util.AuthContextHolder;
import com.atguigu.tingshu.model.user.UserInfo;
import jakarta.servlet.http.HttpServletRequest;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Component;
import org.springframework.util.StringUtils;
import org.springframework.web.context.request.RequestAttributes;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;

/**
 * @author atguigu-mqx
 * @ClassName GuiGuLoginAspect
 * @description: TODO
 * @date 2023年10月27日
 * @version: 1.0
 */
@Component
@Aspect
public class GuiGuLoginAspect {

    @Autowired
    private RedisTemplate redisTemplate;
    //  aop 思想：在不改变原有业务基础上，添加新的业务逻辑. aop：动态代理  ioc：反射
    //  spring 常用的设计模式： 单例，工厂，代理，模版
    //  通知：切点：切面：
    @Around("execution(* com.atguigu.tingshu.*.api.*.*(..)) && @annotation(guiGuLogin)")
    public Object loginAspect(ProceedingJoinPoint point,GuiGuLogin guiGuLogin) throws Throwable {
        //  1.  登录成功的时候，我们生成了一个token，并将token 写入了header中. 因此要获取到请求对象，同时登录成功之后，
        //  我们将用户数据存储到缓存中. 缓存的key 由token 组成.
        RequestAttributes requestAttributes = RequestContextHolder.getRequestAttributes();
        //  获取到ServletRequestAttributes
        ServletRequestAttributes servletRequestAttributes = (ServletRequestAttributes) requestAttributes;
        //  获取到请求对象
        HttpServletRequest request = servletRequestAttributes.getRequest();
        //  从请求头中获取token
        String token = request.getHeader("token");
        //  判断当前用户是否需要登录.
        if (guiGuLogin.required()){
            //  必须要登录。
            if (StringUtils.isEmpty(token)){
                //  说明访问的业务需要登录，但是用户没有登录.
                throw new GuiguException(ResultCodeEnum.LOGIN_AUTH);
            }
            //  从缓存中获取数据 get key value;
            UserInfo userInfo = (UserInfo) this.redisTemplate.opsForValue().get(RedisConstant.USER_LOGIN_KEY_PREFIX + token);
            //  判断
            if (null == userInfo){
                //  说明缓存中没有数据
                throw new GuiguException(ResultCodeEnum.LOGIN_AUTH);
            }
        }

        //  判断token不为空！
        if (!StringUtils.isEmpty(token)){
            //  从缓存中获取数据 get key value;
            UserInfo userInfo = (UserInfo) this.redisTemplate.opsForValue().get(RedisConstant.USER_LOGIN_KEY_PREFIX + token);
            //  判断
            if (null != userInfo){
                //  存储用户信息数据
                AuthContextHolder.setUserId(userInfo.getId());
                AuthContextHolder.setUsername(userInfo.getNickname());
            }
        }
        // start stopwatch
        Object retVal = point.proceed();
        // stop stopwatch
        return retVal;
    }
}
